package treinamentoClassificacao.cln.cdp;

import java.io.FileInputStream;
import java.io.IOException;
import java.io.ObjectInputStream;

import weka.classifiers.functions.MultilayerPerceptron;
import weka.core.Instance;

public class ClassificadorTipo {

	private MultilayerPerceptron mlpTipo = null;
	
	public ClassificadorTipo(String nomeArquivoClassificador) throws IOException, ClassNotFoundException {
		FileInputStream arquivo = new FileInputStream(nomeArquivoClassificador);
		
		ObjectInputStream oisTipo = new ObjectInputStream(arquivo);
		mlpTipo = (MultilayerPerceptron) oisTipo.readObject();
		
		oisTipo.close();
		arquivo.close();
	}
	
	public int classificarDados(double[] dados) throws Exception {
		Instance ins = new Instance(0, dados);
		return (int) mlpTipo.classifyInstance(ins);
	}
}
